// TrainRequest.java
package com.ruoyi.insect.domain;

import lombok.Data;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;


/**
 * 训练表单对象
 */
public class TrainForm {

    private String taskName;
    private String description;
    private Integer epochs;
    private Double learningRate;
    private Integer batchSize;
    private String modelType;
    private Boolean usePretrained;

    public String getTaskName() {
        return taskName;
    }

    public void setTaskName(String taskName) {
        this.taskName = taskName;
    }

    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public Integer getEpochs() {
        return epochs;
    }

    public void setEpochs(Integer epochs) {
        this.epochs = epochs;
    }

    public Double getLearningRate() {
        return learningRate;
    }

    public void setLearningRate(Double learningRate) {
        this.learningRate = learningRate;
    }

    public Integer getBatchSize() {
        return batchSize;
    }

    public void setBatchSize(Integer batchSize) {
        this.batchSize = batchSize;
    }

    public String getModelType() {
        return modelType;
    }

    public void setModelType(String modelType) {
        this.modelType = modelType;
    }

    public Boolean getUsePretrained() {
        return usePretrained;
    }

    public void setUsePretrained(Boolean usePretrained) {
        this.usePretrained = usePretrained;
    }
}
